import numpy as np

#矩阵点乘的方式
def calculateBynp():
    # 单价
    x = np.array([[1.2, 1.5, 1.8],
        [1.3, 1.4, 1.9],
        [1.1, 1.6, 1.7]])
    # 购买数量
    y = np.array([5, 10, 9]).T
    
    price=np.dot(x,y)
    return price


# 计算蔬菜价格
def calculateByFor():
    # 单价
    x = np.array([[1.2, 1.5, 1.8],
        [1.3, 1.4, 1.9],
        [1.1, 1.6, 1.7]])
    # 购买数量
    y = np.array([5, 10, 9]).T
    totalPrice=[] # 采购总价
    for arr in x:
        price=0
        n=0
        for p in arr:
            price+= p * y[n]
            n+=1
        totalPrice.append(price)
    return totalPrice

def main():
    print("\n=====矩阵点乘的方式计算总价======\n")
    totalPrice = calculateBynp()
    print(totalPrice)

def main_for():
    print("\n=====for循环计算总价======\n")
    totalPrice = calculateByFor()
    print(totalPrice)


if __name__=="__main__":

    # 矩阵点乘的方式
    main()

    #for循环计算总价
    main_for()

